import os

from autogen import AssistantAgent,UserProxyAgent


config_list = [{
    "model":"gpt-3.5-turbo",
    "api_key":os.environ["OPENAI_API_KEY_ZHIHU"],
    "api_base":os.environ["OPENAI_API_BASE_ZHIHU"]
}]

assistant = AssistantAgent("assistant",llm_config={"config_list":config_list})
user_proxy = UserProxyAgent("user_proxy",human_input_mode="ALWAYS",code_execution_config={"work_dir":"autogen_work_dir"})

user_proxy.initiate_chat(assistant,message="Plot a chart comparing the year-to-date(YTD) stock price changes of NVIDIA(NVDA) and Tesla.")
